Define functions in Python